import itertools
# count()会创建一个无限的迭代器,从1开始
natuals = itertools.count(1)
for n in natuals:
     print(n)
     if n >= 10:
         break
# cycle()会把传入的一个序列无限重复下去
cs = itertools.cycle('ABC')
for idx, c in enumerate(cs):
    print(idx, c)
    if idx >= 9:
        break
print('------------------')
# repeat()负责把一个元素无限重复下去，不过如果提供第二个参数就可以限定重复次数
ns = itertools.repeat('A', 3)
for idx, n in enumerate(ns):
    print(idx, n)

# 无限序列只有在for迭代时才会无限地迭代下去，如果只是创建了一个迭代对象，它不会事先把无限个元素生成出来，事实上也不可能在内存中创建无限多个元素。
# 无限序列虽然可以无限迭代下去，但是通常我们会通过takewhile()等函数根据条件判断来截取出一个有限的序列：
natuals = itertools.count(1)
ns = itertools.takewhile(lambda x: x <= 10, natuals)
print(list(ns))

# chain()可以把一组迭代对象串联起来，形成一个更大的迭代器：
for c in itertools.chain('ABC', 'XYZ'):
    print(c)
    
print('------------------')
# groupby()把迭代器中相邻的重复元素挑出来放在一起：   
for key, group in itertools.groupby('AAABBBCCAAA'):
    print(key, list(group))
# groupby()的key参数是一个函数，它会根据key函数返回的值把迭代器中的元素分组。
for key, group in itertools.groupby('AaaBBbcCAAa', lambda c: c.upper()):
    print(key, list(group))